# -*- coding: utf-8 -*-
"""
 批处理加水印脚本
"""

import os
import hide
import trans


def ImbedBatch(path,message):
    files=os.listdir(path)
    out="Out"
    if not os.path.exists(out):
        os.makedirs(out)
    s=[]
    ListKey=[] # key列表 保存为txt
    for file in files:
        if not os.path.isdir(file):
            s.append(os.path.join(path,file))
    for file in s:
        print("##### 处理jpg: {} ######:".format(file))
        # 批处理
        jpeg=hide.jio.read(file)
        Key,jpeg=hide.imbed(jpeg,message)
        ListKey.append([Key,"Imbeded"+os.path.basename(file)])
        hide.jio.write(jpeg,os.path.join(out,"Imbeded"+os.path.splitext(os.path.basename(file))[0]+".jpg"))

    hide.ListSave(ListKey,os.path.join(out,"key.txt"))

if __name__=="__main__":
    """
        输入需要加密的水印的目录
        添加水印后，自动另存为"Out"文件夹下
    """
    print("输入需要批量添加加水印图片的目录:")
    path=input()
    print("输入需要批量添加的水印")
    message=input()
    # message="Test"
    message=trans.Str_encode(message)
    ImbedBatch(path,message)
